Minor Changes to main activity