Add class for database operations