Got the flow solver sort of working... Tuned the weights as well