Updated usage for new API