(STAT2401) assignment 2 update