Added the database connection to the class definition.