We have refactored mbuild's box code contributed by @justinGilmer to a separate Python package (available here). It might be nice to extend the current Box implementation by wrapping the functionality using the Box class provided by molbox.