Improves drawing code and removes matplotlib dependancy.

Still requires Network X