Adding return statement