Trying to add class