Code style
Fixed code snippet for small files