(STAT2401) assignment 1 - draft update