Added missin unit test
Minor changes to README