Can you also add a couple tests with nested functions:

1. Where only the inner function has a return. 
2. Where only the outer function has a return. 

These are the trickier cases that are important to get working.