Update MainActivity.java

Add the remove update method so that we can get rid of the both case