all tests passing but need to implement the 4th problem