Add checkbox toggle code