adding .h
change main.cpp