fixing a bug; small layout update