Rename functions, classes, and modules for clarity + mjx bug fix.

Rename functions, classes, and modules for clarity + mjx bug fix. #132