Rework demo with animation, impement Prim's algorithm.
Refactor to remove some redundant code.