Adding some basic tests for prime number checker.
Clean up solution.