Improved the readme
Unit test.