Update README and add the functionality we were after