Instead of setting up few colors you can make an input dialog where the player will choosw the first color id and then the second one. Therefore the script itself is nice.
I thought about that option but I decided not to do it for now as some are lazy (Like me) and rather not go online for the colors, I can add that as an option though in the next version, so it's all good.