Now handles unconditional return statement at end of selected code. Except for the return() added