added test cases for milestone 1