Added some extra tests to problem2.rkt