Cleanup code for deploy