Fix layout on smaller window widths