Clean up toolbox.py