Update ui.py

import argparse