Added the class to connect to the DB