updated class

updated class