changed switch in main.cpp