Updated solve function call