Fix stuff & add `edit` button