Move code from Main to getFireworkColor