Added a unit test file, improved readme