added header file & main