small changes to py