Gui + bug fix