An update to our Assignment 2