Small fix and update readme