Added button to the setup screen
Done button now working