Add $size support