Add assertion tests for problem 2 and and add undefined cases for problem 2