finalised header file for Assign2