updates to Sage Assignment 1