Added GUI code