add tests for lab 6