Repeated Code In Existing File

**Describe the bug**
When new code is added to an existing source file it includes the import and logging statements from the top of the file causing errors. It should just include the new code and strip out any existing headers.