updated class with new methods