Adding unit tests

Signed-off-by: Kevin Kinnebrew <kevin.kinnebrew@gmail.com>