Implemented Unit of Work
Added new database models