You were right, there should be `return` statements where you put them.