Added visualization of sample / reward

.