Added edit mode button