Lab4 is OK, but need more tests