updated code for pollutant assignment 3