Refactor/ui

Separated out UI into specific python modules