Improvement

Added input arguments.