Improve Collapse on mobile