Add code to make sure users don't lose their work by accident