Updated ColorSquarePicker.swift