Updated after HW1.