Corrected the expand_comment function