fix some alignment issues with the vizualisation, suggestions from @taneemt