Last task from Classes part 2