Reused the same Input data type for Stream generator
Refactoring

  Improved readability