Added UnitTest function

Completely passes