Adding tests

ditto