Adds support for return statements in subs. Typechekcing for returns is still missing