Place restore button after current state button

Fixes #9