updated unit test and readme