Adding single epoch python scripts