write tests, update readme