Adjusting and adding to GUI
Added vizualization feature