update python script for new setup with short