The are other places elsewhere that asume one numerical answer. It wouldn't hurt to indicate all the places those assumptions are made with a TODO comment, so we can make the corrections later if we decide to make the module more generic we can easily find the places where changes should be made.