Assignment 5 quick fix