Use Volume class per TODO