Added gradient visualization