problem b py