refinements to assignment 1

refinements to assignment 1