[CHORE] Cleaned up JavaScript wrapper