updated return statement