[FEATURE] Clean up tests, update README.md