Improved code clarity
Updated README