Update readme and add test