Update difficulty.cpp