Fix issue w/ assigning colors fixes #241

Remove this part, as discussed. Merge issue.