Tweak collapse JS