UI enhancements

Signed-off-by: Simon Marchi <simon.marchi@polymtl.ca>