Added Tests

Added Unit Tests