python code should work now