Replaced imageData with fillRect

and removed incorrect color segmentation
Updated TODO since we're not rendering the same way