Add missing files to python driver

Review 881 by @wmrowan
Fixes #1359