updated code/plot