Added button funtinality on main screen