Add a return statement