Updated the layout for boxes