quick fix for project 4