Add support for '@' prefix

Thanks @kevinsawicki. That worked like a charm. I expected a normal JS regex like `@?TODO` to work. It did not :).