Add support for keep special comments. Fixes #3