Defining Classes - Part 2

some changes to task 4.