Using toggle class is much simpler