Small layout update