Add gradient for part 1, forgot that one