add functionality for solution method