Sample code improvements in readme