adding colors to visualization