update tests, clean up code
1.0.1