working on add class