Like you said it would be a simple modification so i would suggest writing it