Added database connection class