Brought back the unit tests

Signed-off-by: Spyros Gasteratos <morfeas3000@gmail.com>