fixed main code snippet in readme