I played around with it for a while and realised it was too complex so after some simplifications it landed in this appearance.