Update a-rainbow.cpp