Refactored problem 3
Add comment to problem 3
created is_prime method