Where to put the next return? I am not able to solve this