Allow to optionally pass device size as the second argument