Add ability to restart loading flow