augment tests complete
Cleanup