Added homework zero code