Modified the main.cpp
Added another function