Added unit testing

PMCP-8 #Done Finished with unittesting