added support for return statements
changed 'return' to 'break'