Added a button to main screen
No big changes