Added visualiser setup