Added entities and created a database connection