adding testing for classes