Updated code in readme reference