Fix README sample code