modify MainActivity class