Both are fine. Include some scenarios/examples, so that someone not sure how to do this understands what they are doing here.