[IMP] Improve memory consumption
Remove pyflakes temporary