Changed hard-coded Color assignments to use a variable instead