Fix for histograms for water dipoles

[MandyHoffmann]

Pre-review checklist for PR author

PR author must check the checkboxes below when creating the PR.

• I've confirmed the contribution guidelines.
• I have reviewed and understand all AI-generated code in this PR.
• I have added human-written tests for the new logic.
• I have properly cited any upstream algorithms or libraries the AI utilized.
• I have disclosed significant AI tool usage in the PR description.

Summary

Reimplemented the histogram colouring.

Linked issue

#759

Testing

Tested on water slab dipole benchmark with trajectories from mace_mp_0a, orb_v3_consv_omol, and pet_mad