Upgrade to Unity 4