add methods to MainScreen.java