Updated readme and fixed a minor bug.