cleaning initialization
initialize colors
fix array indexing
reformat lambda