little UI improovment