part 3
added missing return