Added class to create a db connection