Added Unit Testing and one simple test.