fixed missing things in class definition