Кнопку подключить несложно -- один из контактов подключается к одной из ног GND на ардуино, другой к какой-нибудь из рядом находящихся ножек, например D2.
Как переделать скетч, к сожалению, уже не подскажу. Уже этим не занимаюсь. Попробуйте спросить на специализированных форумах, например, на...