Some work on visualisation

see readme