Remove tests so I can write them in demo