Allow user to specify kwargs as well
Updated README with example