Fix lint issues and documentation