Small bug fix.
Updated README