Add GUI functionality