Unit Test addition

Added Unit Tests