Add Unit Tests

Add Unit Test.

Look into potential testing frames works.

Edited by Aaron Dees