Added Models and Business Logic