Added add class and remove class methods to elements