Changes to the input arg
Improved README