task 1 - visualization code added