added to readme and fixed minor bug