Polished up gui a bit