Updated readme and added some test
Fix tests