Fixed a bit of the main activity