Updated setup.py with the new packages ui and models