Now supports returning inside of ifs and blocks