Add state to all blend tree functions