removed old testing code from __main__