Added a DB connection class