So taken photos of current controller ( cant read anything from it) but i think its 200w 24v and the wires coming off it,
then wires coming off bike,
as far as i understand it ( correct me if i'm wrong - its kinda what i worked out/assumed)
red and black and power go to battery,
yellow, green...