Added restart function, no idea if it works.