small code cleanup
Updated README