Updated code example in readme