Update write function
Keep as test