Got assignment progfun-forcomp working