* Fixes black failures from Previous PR * Fixes equality testing alert * Fixes call to main() alert * Fixes unused import