class functions
small changes in class