updated readme and fixed small bug