cleanup
Merge pull request #5 from rohitgirdhar/master

Cleanup and visualization improvement