Fixing code-style
Fixing copy-pasta