Add colour picker code