updated code in readme