update on python code