Not necessary so feel free to close if you don't want it, but I made some minor improvements to prime_check.py by shortening it a bit and fixing some of the comments/docstring