change in step6.cpp