bug fixes and updated readme