Add callback after commenting

Possible solution to #21