fixing up gui for demo