Added transition functionality to the 2d screens