UI improvement

by Fredrik Zirath