Adds augment/functions, tests and docs