fix for add class