Added some GUI code