Change the database connection, add models