Put edit button on first screen