Expand the tree part is done
remove urgency.py, add exception handling