Added return statement in part 2