Add retry button to error screen