Adding support for custom file sizes