Add some draft tests