Partially fixed bugs

Added a new media query.