Update rose_hw2_main.cpp