added directories
and main.cpp edits