Update pb.py