Update readme to fix code