Updating README and fixing small bug.