Add some classes on Model

Add some class on Model