slight layout update