Tidied up restart script, added comments