Added button + code. Working on issue #10.