updating classes to pass the tests