Simple PyTorch working neural network. Beginner Friendly!