changed the code to pass