Added media query on height