Add class for save modal