Remove useless code
Support sizes for Python tool