Fix assignment 4