Added a sample code for unit testing purpose