Added getSize function