Collapse is now working
New better solution for collapse
Kinda working, but very ugly