Test for return statements