add tests for lab 3-1