Added addClass() function