GUI Bug fixes