Edited GUI abit