GUI fixes