Add counter code
Move augment to own lib file
Release patch 0.0.1