Add class definition for image submission