Corrected assignment 2 file