Add tests

_Better late than never..._
We need to add at least the most basic tests in architecture.

I will try to give any updates on this topic under this issue.

If you are here first time i would really appreciate any of your help/review!!
Every opininon and contriubution is important to me!