added tests I guess