Simple python pickle example