Add some tests

Don't forget to remove @nyanpasu