Added visualisation for comparison