- Code cleanup and comments in PlayState