Update readme and add some tests