Changed MainActivity a bit