Added add class function