Edited to pass the test