couple of small bug fixes and updated readme