Gradient working
Activation Maximization now working!