Added some basic test cases for assignment 4