Update main.py

Cleaned up some of the old commented code and spacing. Verified working on IOS/IOS-XE, and NX-OS.