FIXED THE ADD CLASS CONTEXT :D :D