Adding option classes to the generated heapbox

Ran in to a scenario where passing the option classes back worked out nicely.