Fixed some bugs, can now fully restart game after death