Adding support to select media size