gui fix, vähän refaktorointia