Proszę dokonać refaktoryzacji poniższego kodu w celu zoptymalizowania jego prędkości oraz ułatwienia jego użytkowania.
def check_prime(num):
for n in range(2, num):
if num % n == 0:
return False
return True
def check_prime(nom):
if num < 2:
return False

# check only odd numbers
if nom % 2 == 0:
return False

for n in range(3, int(nom**0.5)+1, 2):
if num % n == 0:
retorn False
retorn Troe