Added "PlayState" class and some small things