enh: HW1 complete in python