Adding restart button