Some updates to match the readme