add input argument support
update readme