Fixed edit button on main view