DRY. Use an array/map?