Fixed Tree issue and Added restart and quit during game