fix CS
fix CS
Merge pull request #429 from deeky666/fix-cs

Fix some more CS