fixed a couple little bugs