Fixed a Couple stupid bugs in the __main__ section.