It certainly possible to add as a manual input parameter. There may also be a built-in parameter for the zone stamp subtype which would allow it to be automatic.
If you plan to write your own, I recommend starting from scratch. The included ones are very complex so as to suit a wide variety of uses. Your requirements are likely quite specific and easily written.