Added GUI functionality