Sample Solution for Unit testing