added solvent_box function