Small fix.
Moved the edit button to the top right corner.