Update readme to use the updated code