Update app to work with latest api class changes