Changed it from multiple files (using bash) to py