Added gradient option