Only show buttons based on the current state
Merge pull request #85 from thescouser89/hide_buttons

Only show buttons based on the current state