adding first tests for solve function