about to change literally everything
Trying to fix gradient check