Add responsive elements and menu toggle for small screens