Added restart funtionality