Removing base code from __main__