Extra credit - add button and message.