Update height on media queries