More visualization. Bug fixes in behaviour. Started work on balancing.