Fixed the solve method to return correct answer