Fix layout for small devices/windows