How to make the form collapse work on mobile