fixed small bug + updated readme