Updated MainActivity's button handling.