some small improvements

Added basic coloring, and took sqrt out of distance check