Cleaned up formatting in __main__