Add GUI code