Refactor code to use pojo instead of local element.
Set the color.