Hot fix - tud fng issue

[dodu94]

Description

This PR solves issue #289 and should be merged after PR #288 Some refactoring was also made of the extract_output() function in the meantime. Additional style modifications come from the black formatting

Fixes #289

Additional dependencies introduced.
N.A.

Type of change

Please select what type of change this is.

• Bug fix (non-breaking change which fixes an issue)
• New benchmark
  • Non-breaking change which entirely uses exisiting classes, structure etc
  • Breaking change which has implemented new/modified classes etc
• New feature
  • Non-breaking change which adds functionality
  • Breaking change fix or feature that would cause existing functionality to not work as expected

Testing

Both CI testing and manual testing have been performed

Checklist:

• My code follows the style guidelines of this project
• I have performed a self-review of my code
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation -> N.A.
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• General testing
  • New and existing unit tests pass locally with my changes
  • Coverage is >80%

[alexvalentine94]

Thanks Davide, tested fix and appears as working. Changes to expoutput neater, likely other places we can do similar.