Improve layout of generated code
Add more documentation to the enum