Updated initial data to also have colors