Attempting to solve gradient issues
Merged back to single exapmle gradients