Updated readme, fixed small bug