https://capressocoffemakerthecheapest.blogspot.com/2023/06/adding-class-objects-to-pytorch.html