https://codetofun.com/python/prime-number/