Fixed imports, now compiles
Experimenting with TorchPlayer