removed colorgrad dependency