added unit test to solutiuon