Made toggle_comment work