Add classes for handling database connection