My fourth assignment for C++