Updated example code and mentioned in readme