Use classList on element directly