class test passing. Need to breakdown more