Add visualisation for genetic algorithm