Fixed for new file structure

Separation of Cuda and Python