Cleanup tests
Add test doc strings