Wrote some test cases for prime.rb