Updated SDK to reflect the new API