Tried to make GPU_corr.py more generalized