Fix color array bug
Finally settled the color assignment thingy. Now to ensure data integrity