Added the database connection class