Assignment 2 quick fix