Fixed bug after cleanup code
Updated readme