Adding tests

surely not ;)