little changes to python script to test device