Add python code used to obtain data