Update MainScreenTest.java