Improved the read me and the code with comments