Add optional size param