Added restart icon functionality