Add class information to factory