fix main.cpp and class function
input can use in unknown line