Removed test code from MainScreen