Add 'collapse' class back after expanding