Задача 'Сверхпростые числа'

Простым числом будем называть натуральное число, большее единицы и делящееся только на единицу и на само себя. Выпишем все простые числа в порядке возрастания и i-ое в этом порядке число обозначим pi (число 2 при этом будет иметь номер 1). Так, например, p1 = 2, p2 = 3, p3 = 5, p52 = 239.

Скажем, что число pi является сверхпростым, если i тоже простое.

Дано натуральное число k. Упорядочим все сверхпростые числа по возрастанию. Найдите k-ое сверхпростое число в этом порядке.

Вот мой код, но тесты не проходит. Можете пожалуйста сказать, что не так?

def simple(x:int):
    count=0
    if x == 1 or x == 0:
        return False
    for i in range(1,x//2+1):
        if x%i==0:
            count+=1
        if count>1:
            return False
    return True
k=int(input())
i=0
numb=0
mas=[]
while i<k+2:
    numb+=1
    if simple(numb):
        i+=1
        if simple(i):
            ends=numb
            mas.append(numb)
print(ends)