Finished code for programming assignment.
Added generated images.