added return in main & updated CMakeLists