added test for return statement