Corrected assignment 1