Add restart callback
Simplify readme