Add cancel button behavior to comments