Fix layout for small and large screens