Started working on incorperating updates to the visualization
merged MainUI into Main