Fix KeyError message
Fix README examples