Moving a servo 90 degrees and then back to its original position with a push of a button

I found the code here:
How to make a servo motor rotate 90 degrees then back to its original position

Arduino Uno

Problem is how to make the code better and help with the wiring.

Thanks but where do I add the button and what voltage power supply?