working on models for the database