Better main script