Small fix and updated README