Adjust media query functionality