polish up gui