Optimize evaluating code.

Generate function and evaluate colours faster.
Need to add code for save screen to file,
pygame.image.save doesn't work.