Update the @module and @class for JS documentation