Code style change
Add a test