slight ui improvements