Some changes in return statement