Added button handling, updated docs