Added some unit tests and fixed the readme file