add class to wrapper