Include kargs in __init__.

Add the required kargs argument in init.