Changed the input param
updated README