Add stuff for class