Copied the code from John Wohl, and updated readme