Adding return comments