Updates to the py-untiwster to use new API