update import in flask example