Simple Pytorch Example