added classes for lab 4, part 1