Fixes for code samples in readme