added size function, updated the respective header file