Fix the NameError
Fixing the error __init__ called with wrong number of arguments