Fix two bugs so example works

Minor bug fixes for README