Added testing for weight solver. Does not assert the correct answer but lets the user visually examine the flows for a simple test case