added name of missing classes in main.cpp