Fixed a bit of GUI