Fixing 04.cpp