My python class works