making python run script better