Improve formatting update v1

[brianhjh]

Description

Updated to add the python module and line number for each unit.

• [ v] New feature (non-breaking change which adds functionality)

Key checklist

• [v] Pre-commit hooks run successfully (eg. pre-commit run --all-files)

Further checks

• [ v] Tests added or an issue has been opened to tackle that in the future. (Indicate issue here: # (issue))