Try to add some animation on expand/collapse