Check in addClass
Fix conditional