tweak script to toggle class