Moved visualisation code to separate directory
Removed unneeded imports
Pinned down and cleaned up dependencies