Added restart function, being awesome and shit