edit button sort of fixed