Redo optimization. Fix print statement.