first attempt at #21 (coloring atoms)