updated for the class