Minor bug fix + README update.