Removing else condition from addClass method

Classes were being overwritten, now fixed.