Working on new gradient transformer