Refactored depth to be a numpy array from the start