Changed to C++, working prompt