Error corrected, checked using CPU results