hw0 code