Moved code into a class and added the code text to the image.