missing parameter

without this, my 'notebook' show errors about missing columns
Merge pull request #1 from danielsodkiewicz/patch-1

missing parameter