Minor update

Bug fix and updated README