WIP - return statement tests