Code formatting tool implemented in Python