Adjusted main() to work with header file