Testing refactor

* Remove unecessary lines
and finally here :)