Make visualisation responsive
Cleanup height and width handling