Added gradient support