Implemented ScreenTransition for smoother transitions between screens
Clean up comments and unused calls