small changes

WARNING: Code won't compile