DOC: Updates the readme and fixes a small bug.