Update Asmt3: bug fix