Thanks, I am using it exactly that way as a separate file in my own folders now, but I was hoping I could modify my torch library to be able to use it for the future. So i will keep it as it is and wont modify main files. But would like to know where is the exposed class list?