feat: edit state, yo
fix: fix disappearing buttons