Adds a few tests I forgot