Got main UI working