Reorganized .py modules. Added flat ui.