MODIFIED: fixed some of the code listing on the README