Fixed a bug related to standalone code blocks
Added a check for the 'return' statement