Update notes for submitted changes

* Fix imports in test_driver