updating classes for case study pg