Make improvements to the script and add documentation
Merge pull request #1 from surabhigupta/improvements

Make improvements to the script and add documentation