Further edited the .h file