Bug fixes + Gui Update