Problem with color_at function