Add some GUI code