Update Problem.rb and Problem_test.rb