Fixed sample code in readme