minor fix
Small changes in the display