updated readme, fixed small bug