added HW0 codes