Better sample code in readme