Add some deploy code + tests