Updated documentation
Corrected documentation and removed __main__ block.