Port process.py to Python 2
Number of steps as feature