updated sample
use logging instead of prints
styling update