added unit tests
updated readme