Fixed bug and updated README