Added GUI funtionality