add the return statement