Adding a return statement