Tidy up MainActivity