Modified GUI slightly