Added toggle comment functionality