[TASK] added first Unit Tests
[FEATURE] Adding some more unit tests :D