hw0 updated program5.r