including import statements in the test generator. (fixes #152)