Fixes compute_gradient func for new images
Renamed directory