Updated assignment week 6.