Changed GUI a bit