Updated to include the new contract