improve controller UI