updating work done in class but not fixed yet