figured out bug
passed all lab 4 tests