Updates screenshot
Adds script for testing color assignments