adding support for custom boxes