Small changes to main activity