missing assignment
renaming variables to match original python code