Commented all the code
Minor modification in the main file