Added specialised model for Class