Fixed code example in readme