functionality is working, just can't get 2 tests to pass