Updated with new Unity Editor class