Update CHS2546 assignment 2