fixing unit tests
milestone lab6 test passes