Fix layout for medium screens