python changes and better UI