Add test for nested functions/ifs/returns.