Minor update to main activity