Add prime.py and unit_test.py.

Kész a házi feladat.