Generated images, need docstrings and comments