fixing warmup script for python3