another improvement in UI :)