Small fix in mini-assignment 3