Feature/testing

-[x] Create tests
-[x] Reach 100% coverage