Add option to override size