Basic AI visualizations working.  Cleaned up code.