cleanup imports
adds a test-case
update README with testing