Addition to small-step code.