Updated size method