Re-add main script