small update to the layout