modifications to viz. Need to tune params.