Added more tests for lab 1