snaz up code samples in readme