Fix a small bug and updated README file