Extend code sample in readme