Tweaked and added comments
Updated readme