Regrouping everything into a single python script