Removed unused import

The math module is not used in this file.