refactor codebase and add new visualisations
small bugfixes and additional work