Fixing GUI part4