Add addClass function