Add comment callback function. Fixes #95