cleaning up memory_usage_formatter.py