Add retry button and loading message

Fixes #1