Added fix and better README