Added database connection framework