made few more changes to header file