Change example code in readme