Fixed broken test, wrote method that passes all tests