Add toggle comment functionality