working on models
Implemented some classes