hw-4: added header file and included it in both files