Updated README and fixed a small bug