Adding tests

leftover, I'll remove it