* Fix main.cpp again