Added support for 'return';