Remove custom classes from haloplex.py