GUI almost working