Add a small bit of documentation
Modernize code