added hw0 code