Added functionality to apply CSS Transition Classes on display toggle