Added test cases

According to lab instructions