Fixed README and small bug