update for progfun-003
complete assignment 2