Solved it but isn't passing tests