Making the run python script more intuitive