CLASS: working on first assignment for progfun003