add more tests for assignment1