Add a few more conditional buttons