tests: Adding tests idea. Unfinished