My first assignment in C++