Improve UI v4