Added return statement test.